import React from 'react'

interface ContainerProps {
  className?: string
  size?: 'normal' | 'big'
}

const Container: React.FC<ContainerProps> = ({className, size, children}): React.ReactElement => {
  size = size ? size : 'normal'
  const classNameResult =
    (size === 'big' ? 'container ' : 'container-' + size + ' ') + (className ? className : '')
  return <div className={classNameResult}>{children}</div>
}

export {Container}
